package cn.myworld.algorithm.linear;


import java.util.Iterator;

public class Stack<T> implements Iterable<T> {
    // 记录首节点
    private Node head;
    // 栈中元素个数
    private int N;

    private class Node {
        public T  data;
        public Node next;

        public Node(T data, Node next) {
            this.data = data;
            this.next = next;
        }
    }

    public Stack() {
        this.head = new Node(null, null);
        this.N = 0;
    }

    // 判断当前栈是否为空
    public boolean isEmpty() {
        return N == 0;
    }

    // 获取栈中元素个数
    public int size() {
        return N;
    }

    // 压栈
    public void push(T t) {
        // 找到首节点指向的第一个节点
        Node oldHead = head.next;
        // 创建新节点
        Node newNode = new Node(t, null);
        // 让首节点指向新节点
        head.next = newNode;
        // 让新节点指向原先第一个节点
        newNode.next = oldHead;
        // 元素个数+1
        N++;
    }

    // 弹栈
    public T pop() {
        // 找到首节点指向的第一个节点
        Node oldHead = head.next;
        if (oldHead == null) {
            return null;
        }
        // 让首节点指向原来第一个节点的下一个节点
        head.next = oldHead.next;
        // 元素个数-1
        N--;
        return oldHead.data;
    }

    public T peek() {
        return head.next.data;
    }

    @Override
    public Iterator<T> iterator() {
        return new SIterator();
    }

    private class SIterator implements Iterator<T> {
        private Node cur;
        public SIterator() {
            cur = head;
        }

        @Override
        public boolean hasNext() {
            return cur.next != null;
        }

        @Override
        public T next() {
            cur = cur.next;
            return cur.data;
        }
    }

}
